home *** CD-ROM | disk | FTP | other *** search
/ Sprite 1984 - 1993 / Sprite 1984 - 1993.iso / src / lib / c / stdio / setvbuf.c < prev    next >
C/C++ Source or Header  |  1989-06-19  |  3KB  |  103 lines
  1. /* 
  2.  * setvbuf.c --
  3.  *
  4.  *    Source code for the "setvbuf" library procedure.
  5.  *
  6.  * Copyright 1988 Regents of the University of California
  7.  * Permission to use, copy, modify, and distribute this
  8.  * software and its documentation for any purpose and without
  9.  * fee is hereby granted, provided that the above copyright
  10.  * notice appear in all copies.  The University of California
  11.  * makes no representations about the suitability of this
  12.  * software for any purpose.  It is provided "as is" without
  13.  * express or implied warranty.
  14.  */
  15.  
  16. #ifndef lint
  17. static char rcsid[] = "$Header: /sprite/src/lib/c/stdio/RCS/setvbuf.c,v 1.4 89/06/19 14:15:14 jhh Exp $ SPRITE (Berkeley)";
  18. #endif not lint
  19.  
  20. #include <stdio.h>
  21. #include <stdlib.h>
  22. #include "fileInt.h"
  23.  
  24. /*
  25.  *----------------------------------------------------------------------
  26.  *
  27.  * setvbuf --
  28.  *
  29.  *    This procedure may be used to amount of buffering used for
  30.  *    a stdio stream.  A buffer size of 0 or 1, or a mode of _IONBF,
  31.  *    means the stream is unbuffered: each byte of I/O results in a
  32.  *    system call.  Needless to say, this is likely to be inefficient.
  33.  *    A buffer size of N means that N characters are read from the
  34.  *    system at a time, regardless of how many are actually needed,
  35.  *    and on writes, nothing is written to the system until N bytes
  36.  *    have accumulated, at which point all are written in one operation.
  37.  *
  38.  * Results:
  39.  *    Returns 0 on success, -1 if there was an error in the arguments
  40.  *    or if buffer space couldn't be setup properly.
  41.  *
  42.  * Side effects:
  43.  *    The stream is modified to use size bytes of buffering.
  44.  *    Old buffer space is returned to the storage allocator, if it
  45.  *    was allocated by stdio rather than the client.
  46.  *
  47.  *----------------------------------------------------------------------
  48.  */
  49.  
  50. int
  51. setvbuf(stream, buf, mode, size)
  52.     register FILE *stream;    /* Stream whose buffering is to be changed. */
  53.     char *buf;            /* Buffer to use for stream. */
  54.     int mode;            /* Mode for buffering:  _IOFBF, _IOLBF, or
  55.                  * _IONBF. */
  56.     int size;            /* Number of characters of space in buffer;
  57.                  * 0 or 1 means unbuffered. */
  58. {
  59.     int result;
  60.  
  61.     if (stream->readProc != (void (*)()) StdioFileReadProc) {
  62.     return -1;
  63.     }
  64.  
  65.     result = fflush(stream);
  66.  
  67.     /*
  68.      * Careful!  Don't free the statically-allocated buffers for
  69.      * standard streams.
  70.      */
  71.  
  72.     if ((stream->buffer != stdioTempBuffer)
  73.         && (stream->buffer != stdioStderrBuffer)
  74.         && (stream->buffer != NULL)
  75.         && !(stream->flags & STDIO_NOT_OUR_BUF)) {
  76.     free((char *) stream->buffer);
  77.     }
  78.     stream->flags &= ~(STDIO_LINEBUF|STDIO_NOT_OUR_BUF);
  79.  
  80.     if (size < 1) {
  81.     size = 1;
  82.     }
  83.     if (mode == _IONBF) {
  84.     size = 1;
  85.     } else if (mode == _IOLBF) {
  86.     stream->flags |= STDIO_LINEBUF;
  87.     } else if (mode != _IOFBF) {
  88.     return -1;
  89.     }
  90.  
  91.     if (buf != 0) {
  92.     stream->buffer = (unsigned char *) buf;
  93.     stream->flags |= STDIO_NOT_OUR_BUF;
  94.     } else {
  95.     stream->buffer = (unsigned char *) malloc((unsigned) size);
  96.     }
  97.     stream->bufSize = size;
  98.     stream->lastAccess = stream->buffer - 1;
  99.     stream->readCount = 0;
  100.     stream->writeCount = 0;
  101.     return result;
  102. }
  103.